package com.qf.scms.controller;

import com.alibaba.excel.EasyExcel;
import com.qf.scms.common.PageVO;
import com.qf.scms.entity.Course;
import com.qf.scms.entity.CourseExcel;
import com.qf.scms.listener.CourseExcelDataListener;
import com.qf.scms.param.CourseParam;
import com.qf.scms.service.CourseService;
import com.qf.scms.utils.R;
import com.qf.scms.vo.CourseVO;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;

/**
 * (Course)表控制层
 *
 * @author makejava
 * @since 2025-06-25 21:17:49
 */
@RestController
@RequestMapping("course")
public class CourseController {
    /**
     * 服务对象
     */
    @Resource
    private CourseService courseService;

    /**
     * 分页查询
     *
     * @param courseParam 查询参数
     * @return
     */
    @GetMapping("/page")
    public R queryByPage(CourseParam courseParam) {
        PageVO<CourseVO> list = this.courseService.queryByPage(courseParam);
        return R.ok(list);
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public ResponseEntity<Course> queryById(@PathVariable("id") Integer id) {
        return ResponseEntity.ok(this.courseService.queryById(id));
    }

    /**
     * 新增课程
     *
     * @param course 实体
     * @return 新增结果
     */
    @PostMapping("/add")
    public R add(@RequestBody Course course) {
        Integer result = this.courseService.insert(course);

        if (result <= 0) {
            return R.fail("添加失败");
        }
        return R.ok("添加成功");
    }

    /**
     * 编辑数据
     *
     * @param course 实体
     * @return 编辑结果
     */
    @PutMapping("/update")
    public R edit(@RequestBody Course course) {
        Integer result = this.courseService.update(course);

        if (result <= 0) {
            return R.fail("修改失败");
        }
        return R.ok("修改成功");
    }

    /**
     * 删除数据
     *
     * @param id 主键
     * @return 删除是否成功
     */
    @DeleteMapping("/delete")
    public R deleteById(Integer id) {
        boolean result = this.courseService.deleteById(id);
        if (!result) {
            return R.fail("删除失败");
        }
        return R.ok("删除成功");
    }

    /**
     * 批量删除数据
     *
     * @param ids 主键
     * @return 删除是否成功
     */
    @DeleteMapping("/deleteBatch")
    public R deleteBatch(@RequestParam List<Integer> ids) {
        boolean result = this.courseService.deleteBatch(ids);
        if (!result) {
            return R.fail("批量删除失败");
        }
        return R.ok("批量删除成功");
    }

    /**
     * 分批量批量导入
     *
     * @param file excel课程数据文件
     * @return R对象
     */
    @PostMapping("/importCourse")
    public R importCourse(MultipartFile file) {
        try {
            // 读取Excel数据，注意这里是默认读取的第sheet0的数据
            EasyExcel.read(file.getInputStream(), CourseExcel.class, new CourseExcelDataListener(courseService)).sheet().doRead();
            return R.ok("导入成功");
        } catch (Exception e) {
            e.printStackTrace();
            return R.fail("导入失败");
        }
    }

    /**
     * 导出课程数据
     *
     * @param response 响应对象
     * @throws IOException io异常
     */
    @GetMapping("/downLoadCourseExcel")
    public void downloadExcel(HttpServletResponse response) throws IOException {

        //  查询数据库全部数据
        List<CourseVO> list = this.courseService.selectAllCourseVO();
        // 导出.xls用这个
        // response.setContentType("application/vnd.ms-excel");
        // 导出.xlsx用这个
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码
        String fileName = URLEncoder.encode("所有课程数据", "UTF-8");
        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");

        // easyexcel导出工具
        EasyExcel.write(response.getOutputStream(), CourseVO.class).autoCloseStream(Boolean.FALSE).sheet("所有课程数据")
                .doWrite(list);
    }


}

